package com.example.lotterysystem.service.impl;

import com.example.lotterysystem.common.errorcode.ServiceErrorCodeConstants;
import com.example.lotterysystem.common.exception.ServiceException;
import com.example.lotterysystem.service.PictureService;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Component;
import org.springframework.web.multipart.MultipartFile;

import java.io.File;
import java.io.IOException;
import java.util.UUID;

@Component
public class PictureServiceImpl implements PictureService {

    @Value("${pic.local-path}")
    private String localPath;

    @Override
    public String savePicture(MultipartFile multipartFile) {

        // 创建目录
        File dir = new File(localPath);
        if (!dir.exists()) {
            dir.mkdirs();
        }

        // 创建索引 aaa.jpg -> xxx.jpg
        // 第一步：需要拿到当前文件的全名称，如：aaa.jpg
        String filename = multipartFile.getOriginalFilename();
        assert filename != null;
        // 第二步：拿到当前文件的后缀名，如：.jpg
        String suffix = filename.substring(filename.lastIndexOf("."));
        // 第三步：生成索引，如：xxx
        // 第四步：拼接 xxx.jpg（拼接为自己管控的文件名）
        filename = UUID.randomUUID() + suffix;

        // 图片保存
        try {
            // transferTo 方法将创建好的 multipartFile 保存到自定义的目录(localPath + "/" + filename)中
            multipartFile.transferTo(new File(localPath + "/" + filename));
        } catch (IOException e) {
            throw new ServiceException(ServiceErrorCodeConstants.PIC_UPLOAD_ERROR);
        }

        return filename;
    }
}